package leetcode_61_80;

import java.util.*;
import java.util.stream.Collectors;

public class combine_77 {
    /**
     * 给定两个整数 n 和 k，返回范围 [1, n] 中所有可能的 k 个数的组合。
     * 你可以按 任何顺序 返回答案。
     * @param n
     * @param k
     * @return
     */
    public List<List<Integer>> combine(int n, int k) {
        /**
         * 循环结构是动态的
         * 思路是递归回溯
         */
        function(n,1,k);
        return ans;
    }
    List<Integer>combine=new ArrayList<>();
    List<List<Integer>>ans=new ArrayList<>();
    /**
     * 递归回溯求所有组合
     * 初始版本
     * 16ms
     * 56%
     * @param n  1-n个数字
     * @param i   当前递归起始数字
     * @param k  还需要k个数字完成递归
     */
    public void function(int n,int i,int k){
        if(k==0){
            //这里因为combine是引用对象，需要将数据复制一份添加
            ans.add(new ArrayList<>(combine));
            return;
        }
        for (int t = i; t <= n; t++) {
            combine.add(t);
            function(n, t+1, k-1);
            combine.remove(combine.size()-1);
        }
    }


    /**
     * 优化剪枝
     * @param n
     * @param k
     * @return
     */
    public List<List<Integer>> combine2(int n, int k) {
        function2(n,1,k);
        return ans2;
    }
    List<Integer> combine2 = new ArrayList<>();
    List<List<Integer>>ans2=new ArrayList<>();
    /**
     * 优化剪枝后
     */
    public void function2(int n,int i,int k){
        if(k==0){
            //这里因为combine是引用对象，需要将数据复制一份添加
            ans2.add(new ArrayList<>(combine2));
            return;
        }
        /**
         * 如果这里后边的元素已经不足够满足一个组合数量，直接直接剪掉
         */
        for (int t = i; t <= n - k +1; t++) {
            combine2.add(t);
            function2(n, t+1, k-1);
            combine2.remove(combine2.size()-1);
        }
    }
}
